package likouJava;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

public class 二叉树的层序遍历 {
    public List<List<Integer>> levelOrder(TreeNode root) {
        List<List<Integer>> res = new ArrayList<>();
        if(root == null) return res;

        //用队列来实现广度优先搜索(LinkedList类实现了Queue接口，因此我们可以把LinkedList当成Queue来用)
        Queue<TreeNode> queue = new LinkedList<>();
        queue.offer(root); //把 root头 放进去

        while (!queue.isEmpty()){
            int levelSize = queue.size();
            List<Integer> currentLevel = new ArrayList<>();

            //遍历当前层
            for (int i = 0; i < levelSize; i++) {
                TreeNode node = queue.poll(); //获取第一个
                currentLevel.add(node.val);

                //将左右子树添加到队列中
                if(node.left != null) queue.offer(node.left);
                if (node.right != null)queue.offer(node.right);
            }
            res.add(currentLevel);
        }
        return res;
    }

}
